package com.ruoyi.common.utils;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Random;

/**
 * @author ：jijinwu
 * @ClassName OrderUtils
 * @date ：2024/7/23 - 22:52
 * @version: V1.0
 * @slogan: 天下风云出我辈，一入代码岁月催
 * @description: 订单编码生成器，生成18位编码，
 * 生成规则 1位单号类型+10位时间戳+7位(用户ID加密&随机数)
 **/
public class OrderUtils {

    // 订单类别头
    private static final String ORDER_CODE = "1";
    private static final String RETURN_ORDER = "2";
    private static final String REFUND_ORDER = "3";

    // 随机数映射
    private static final int[] r = new int[]{7, 9, 6, 2, 8, 1, 3, 0, 5, 4};

    // 用户ID和随机数总长度
    private static final int MAX_LENGTH = 7; // 包括用户ID和随机数

    /**
     * 根据用户ID进行加密并加上随机数组成固定长度编码
     *
     * @param userId 用户ID
     * @return 加密后的编码
     */
    private static String toCode(Long userId) {
        String idStr = userId.toString();
        StringBuilder idBuilder = new StringBuilder();
        for (int i = idStr.length() - 1; i >= 0; i--) {
            idBuilder.append(r[idStr.charAt(i) - '0']);
        }
        // 添加随机数补全长度
        idBuilder.append(getRandom(MAX_LENGTH - idStr.length()));
        return idBuilder.toString();
    }

    /**
     * 生成时间戳（10位）
     *
     * @return 时间戳
     */
    private static String getDateTime() {
        DateFormat sdf = new SimpleDateFormat("yyyyMMddHH");
        return sdf.format(new Date());
    }

    /**
     * 生成固定长度随机码
     *
     * @param n 长度
     * @return 随机数
     */
    private static String getRandom(int n) {
        Random random = new Random();
        StringBuilder randomNumber = new StringBuilder();
        for (int i = 0; i < n; i++) {
            randomNumber.append(random.nextInt(10));
        }
        return randomNumber.toString();
    }

    /**
     * 生成编码（不带类别标头）
     *
     * @param userId 用户ID
     * @return 生成的编码
     */
    private static synchronized String getCode(Long userId) {
        userId = userId == null ? 10000L : userId;
        return getDateTime() + toCode(userId);
    }

    /**
     * 生成订单单号编码
     *
     * @param userId 用户ID
     * @return 订单单号
     */
    public static String getOrderCode(Long userId) {
        return ORDER_CODE + getCode(userId);
    }

    /**
     * 生成退货单号编码
     *
     * @param userId 用户ID
     * @return 退货单号
     */
    public static String getReturnCode(Long userId) {
        return RETURN_ORDER + getCode(userId);
    }

    /**
     * 生成退款单号编码
     *
     * @param userId 用户ID
     * @return 退款单号
     */
    public static String getRefundCode(Long userId) {
        return REFUND_ORDER + getCode(userId);
    }

}
